from files.manager import analyze_watchlist
from . import custommodule


def menu():
    while True:
        print("\n" + "=" * 20)
        print(" 股票诊断菜单 ")
        print("=" * 20)
        print("1. 自选股诊断")
        print("2. 推荐股诊断")
        print("3. 指定股诊断")
        print("0. 返回主菜单")
        print("-" * 20)
        choice = input("请输入选项（0-3）：").strip()

        # 处理用户选择
        match choice:
            case '1':
                stock_codes = [code[0] for code in custommodule.get_watchlist(1)]
            case '2':
                stock_codes = [code[0] for code in custommodule.get_watchlist(2)]
            case '3':
                stock_codes = input("请输入指定股票代码（多个股票代码用空格分隔）：").split()
            case '0':
                break
            case _:
                print("⚠️ 输入错误，请输入 0-3 之间的数字")

        # 显示诊断结果
        result = analyze_watchlist(stock_codes)
        if result.empty:
            print("⚠️ 股票代码或诊断结果异常！")
        else:
            print(result)
